# -*- coding: utf-8 -*-
# @Time    : 2020/9/6 下午5:13
# @Author  : lilong

"""
以最小的数据训练一个机器人的core部分
"""

from rasa.train import train_core

# 需要的文件
config_file = "../bot_playground/bot_tiny/config.yml"
train_files = "../bot_playground/bot_tiny/data"
domain_file = "../bot_playground/bot_tiny/domain.yml"

output_path = "../bot_playground/bot_tiny/models/core"
# train_path = "../bot_playground/tmp_dir"

train_path = train_core(
    domain=domain_file,
    config=config_file,
    stories=train_files,
    output=output_path,
)

print("train_path:", train_path)
